import torch

a = torch.rand(2, 3)
b = torch.rand(3)

print("a =", a)
print("b =", b)

c = a + b
print("c =", c)